package cn.wolfcode.trip.base.util.websocket;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.socket.server.standard.SpringConfigurator;

import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

/**
 * Websocket class
 *
 * @author lishuanwang
 * @date 2019/5/31
 */
/**
 * websocket连接URL地址和可被调用配置
 * @author lishuanwang
 */
@ServerEndpoint(value="/websocketDemo/{userId}",configurator = SpringConfigurator.class)
public class Websocket {
    /**
     * 日志记录
     */
    private Logger logger = LoggerFactory.getLogger(Websocket.class);
    /**
     * 静态变量,记录当前的用户数,线程安全
     */
    public static int onlineCount = 0;
    /**
     * 记录每一个用户的终端, map也要使用线程安全的.
     */
    private static Map<Long, Set<Websocket>> userSocket = new HashMap<>();
    /**
     * 需要session对用户发送数据
     */
    private Session session;
    /**
     * 需要用户的id,记录信息的接受者.
     */
    private Long userId;

    /**
     * websocket连接的session属性,onopen:websocekt连接建立时的操作
     * @param userId 用户的id
     * @param session  session websocket连接的session属性
     */
    @OnOpen
    public void onOpen(@PathParam("userId") Long userId, Session session){
        this.session = session;
        this.userId = userId;
        onlineCount++;
        if (userSocket.containsKey(userId)){
            logger.debug("当前用户id:{}已有其他终端登录"+userId);
            userSocket.get(userId).add(this);
        }else {
            logger.debug("当前的用户id:{}第一次登陆", userId);
            Set<Websocket> addUserSet = new HashSet<>();
            addUserSet.add(this);
            userSocket.put(userId, addUserSet);
        }
        logger.debug("用户{}登录的终端个数是为{}",userId,userSocket.get(this.userId).size());
        logger.debug("当前在线用户数为：{},所有终端个数为：{}",userSocket.size(),onlineCount);
    }


    @OnClose
    public void onClose(){
        //移除当前用户终端登录的websocket信息,如果该用户的所有终端都下线了，则删除该用户的记录
        if (userSocket.get(this.userId).size() == 0){
            userSocket.remove(this.userId);
        }else{
            userSocket.get(this.userId).remove(this);
        }
        logger.debug("用户{}登录的终端个数是为{}",userId,userSocket.get(this.userId).size());
        logger.debug("当前在线用户数为：{},所有终端个数为：{}",userSocket.size(),onlineCount);
    }

    /**
     * @Title: onMessage
     * @Description: 收到消息后的操作
     * @param @param message 收到的消息
     * @param @param session 该连接的session属性
     */
    @OnMessage
    public void onMessage(String message, Session session) {
        logger.debug("收到来自用户id为：{}的消息：{}",this.userId,message);
        if(session ==null) {
            logger.debug("session null");
        }
    }

    /**
     * @Title: onError
     * @Description: 连接发生错误时候的操作
     * @param @param session 该连接的session
     * @param @param error 发生的错误
     */
    @OnError
    public void onError(Session session, Throwable error){
        logger.debug("用户id为：{}的连接发送错误",this.userId);
        error.printStackTrace();
    }

    /**
     * 发送消息
     */
    public Boolean sendMessage(Long receiverId,String message){
        if (userSocket.containsKey(receiverId)){
            logger.debug(" 给用户id为：{}的所有终端发送消息：{}",receiverId,message);
            for (Websocket ws: userSocket.get(receiverId)){
                logger.debug("sessionId为:{}",ws.session.getId());
                try {
                    ws.session.getBasicRemote().sendText(message);
                } catch (IOException e) {
                    e.printStackTrace();
                    logger.debug(" 给用户id为：{}发送消息失败",receiverId);
                    return false;
                }
            }
            return true;
        }

        logger.debug("发送错误：当前连接不包含id为：{}的用户",receiverId);
        return false;
    }

}
